import pygame
import plane_sprites


class PlaneGame:
    """飞机大战主游戏"""

    def __init__(self):
        print("游戏初始化")

        # 1.创建游戏窗口,size是一个元组包含了宽度和高度
        self.screen = pygame.display.set_mode(plane_sprites.SCREEN_RECT.size)

        # 2.创建游戏时钟
        self.clock = pygame.time.Clock()

        # 3.调用私有方法，创建精灵和精灵组
        self.__create_sprites()

        # 4.设置定时器事件
        pygame.time.set_timer(plane_sprites.CREATE_ENEMY_EVENT, 1000)
        pygame.time.set_timer(plane_sprites.HERO_FIRE_EVENT, 150)

    # 创建精灵
    def __create_sprites(self):

        # 创建背景精灵
        background1 = plane_sprites.Background()
        background2 = plane_sprites.Background(True)

        self.back_group = pygame.sprite.Group(background1, background2)

        # 创建敌机的精灵组
        self.enemy_group = pygame.sprite.Group()

        # 创建英雄飞机的精灵和精灵组
        self.hero = plane_sprites.Hero()
        self.hero_group = pygame.sprite.Group(self.hero)

    def start_game(self):
        print("游戏开始")

        while True:
            # 1.设置刷新帧率
            self.clock.tick(plane_sprites.FRAME_PER_SEC)

            # 2.事件监听
            self.__event_handler()

            # 3.碰撞检测
            self.__check_collide()

            # 4.更新绘制精灵组
            self.__update_sprites()

            # 5.更新显示
            pygame.display.update()

            pygame.event.get()

    # 事件监听
    def __event_handler(self):

        for event in pygame.event.get():

            # 判断是否退出游戏
            if event.type == pygame.QUIT:
                self.__game_over()
            elif event.type == plane_sprites.CREATE_ENEMY_EVENT:
                # print("敌机出场")

                # 创建敌机精灵
                enemy = plane_sprites.Enemy()

                # 将敌机精灵添加到敌机精灵组
                self.enemy_group.add(enemy)

            elif event.type == plane_sprites.HERO_FIRE_EVENT:
                self.hero.fire()

        # 使用键盘提供的方法获取键盘按键 - 按键元组
        key_pressed = pygame.key.get_pressed()
        # 判断元组中对应的按键索引值 1
        if key_pressed[pygame.K_RIGHT]:
            self.hero.speed = 2

        elif key_pressed[pygame.K_LEFT]:
            self.hero.speed = -2

        else:
            self.hero.speed = 0

    # 碰撞检测
    def __check_collide(self):

        # 1.子弹摧毁敌机
        pygame.sprite.groupcollide(self.hero.bullets, self.enemy_group, True, True)

        # 2.敌机摧毁英雄飞机
        enemies = pygame.sprite.spritecollide(self.hero, self.enemy_group, True)

        # 3.判断是否列表有内容
        if len(enemies) > 0:

            # 让英雄牺牲
            self.hero.kill()

            # 结束游戏
            PlaneGame.__game_over()

    # 更新绘制精灵组
    def __update_sprites(self):

        self.back_group.update()
        self.back_group.draw(self.screen)

        self.enemy_group.update()
        self.enemy_group.draw(self.screen)

        self.hero_group.update()
        self.hero_group.draw(self.screen)

        self.hero.bullets.update()
        self.hero.bullets.draw(self.screen)

    # 退出游戏
    def __game_over(self):
        print("游戏结束...")

        pygame.quit()
        exit()


if __name__ == '__main__':

    # 创建游戏对象
    game = PlaneGame()

    # 启动游戏
    game.start_game()

